# llms/http/llm_env.py
from db.models.llm_profile import crud

"""
默认为智谱，
ID：
1.智谱
"""
DEFAULT_LLM_MODE_ID = 1

DEFAULT_BASE_URL = crud.get_llm_model_by_id(llm_model_id=DEFAULT_LLM_MODE_ID)["base_url"]


def init_base_url(base_url: str = None):
    if base_url:
        return base_url
    else:
        return DEFAULT_BASE_URL


def init_api_key(api_key: str = None):
    if api_key:
        return api_key
    else:
        raise ValueError(f"API key not found in the database.")
